
# ########## seed 100 ############

# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method RNNSAC \
#     --buffer_size 100000 \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 100 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method OURSACFastAll \
#     --buffer_size 100000 \
#     --encoder_eps_type all \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 100 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method OURSACFastOne \
#     --buffer_size 100000 \
#     --encoder_eps_type all \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 100 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &



# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method OURSACFastBase \
#     --buffer_size 100000 \
#     --encoder_eps_type all \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 100 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


# ########## seed 101 ############


# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method RNNSAC \
#     --buffer_size 100000 \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 101 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method OURSACFastAll \
#     --buffer_size 100000 \
#     --encoder_eps_type all \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 101 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method OURSACFastOne \
#     --buffer_size 100000 \
#     --encoder_eps_type all \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 101 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &



# CUDA_VISIBLE_DEVICES=0 nohup python main.py \
#     --env_name HopperEnv \
#     --env_hook DominoHook \
#     --use_weighted_info_nce \
#     --method OURSACFastBase \
#     --buffer_size 100000 \
#     --encoder_eps_type all \
#     --train_freq 32 \
#     --gradient_steps 32 \
#     --learning_rate 1e-4 \
#     --contrast_frame_stack 10 \
#     --seed 101 \
#     --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
#     --test_eps_num_per_env 10 \
#     --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &




########## seed 102 ############


CUDA_VISIBLE_DEVICES=1 nohup python main.py \
    --env_name HopperEnv \
    --env_hook DominoHook \
    --use_weighted_info_nce \
    --method RNNSAC \
    --buffer_size 100000 \
    --train_freq 32 \
    --gradient_steps 32 \
    --learning_rate 1e-4 \
    --contrast_frame_stack 10 \
    --seed 102 \
    --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
    --test_eps_num_per_env 10 \
    --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


CUDA_VISIBLE_DEVICES=1 nohup python main.py \
    --env_name HopperEnv \
    --env_hook DominoHook \
    --use_weighted_info_nce \
    --method OURSACFastAll \
    --buffer_size 100000 \
    --encoder_eps_type all \
    --train_freq 32 \
    --gradient_steps 32 \
    --learning_rate 1e-4 \
    --contrast_frame_stack 10 \
    --seed 102 \
    --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
    --test_eps_num_per_env 10 \
    --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &


CUDA_VISIBLE_DEVICES=1 nohup python main.py \
    --env_name HopperEnv \
    --env_hook DominoHook \
    --use_weighted_info_nce \
    --method OURSACFastOne \
    --buffer_size 100000 \
    --encoder_eps_type all \
    --train_freq 32 \
    --gradient_steps 32 \
    --learning_rate 1e-4 \
    --contrast_frame_stack 10 \
    --seed 102 \
    --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
    --test_eps_num_per_env 10 \
    --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &



CUDA_VISIBLE_DEVICES=1 nohup python main.py \
    --env_name HopperEnv \
    --env_hook DominoHook \
    --use_weighted_info_nce \
    --method OURSACFastBase \
    --buffer_size 100000 \
    --encoder_eps_type all \
    --train_freq 32 \
    --gradient_steps 32 \
    --learning_rate 1e-4 \
    --contrast_frame_stack 10 \
    --seed 102 \
    --test_envs "[([0.25, 0.375], [0.25, 0.375]),([0.25, 0.375], [1.75, 2.0]),([1.75, 2.0], [0.25, 0.375]),([1.75, 2.0], [1.75, 2.0])]" \
    --test_eps_num_per_env 10 \
    --train_envs "[([0.5, 0.75, 1.0], [0.5, 0.75, 1.0]),([0.5, 0.75, 1.0], [1.25, 1.5]),([1.25, 1.5], [0.5, 0.75, 1.0]),([1.25, 1.5], [1.25, 1.5])]" &

